from enum import Enum


class StrategyType(Enum):
    VOLUME = ("交易量", "VOLUME")
    PRICE = ("交易价", "PRICE")
    NEWS = ("消息面", "NEWS")
    TIME_POINT = ("时间节点", "TIME_POINT")
    COMPREHENSIVE = ("综合", "COMPREHENSIVE")

    def __init__(
        self,
        code: str,
        description: str,
    ):
        self.description = description  # 中文描述
        self.code = code  # Code 标识

    def __str__(self):
        return f"{self.code}: {self.description}"
